mujoco-py==2.1.2.14
tqdm==4.66.1
flax==0.7.5
numba==0.59.0
optax==0.1.7
wandb==0.16.1
tensorflow-probability==0.23.0
seaborn==0.13.2
matplotlib==3.8.2
absl-py==2.0.0
pandas==2.1.4
gdown==5.1.0
ml-collections==0.1.1
scipy==1.11.4
distrax==0.1.5
orbax-checkpoint==0.4.8